Virtual reality based distraction